import os
import sys
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
import pandas as pd
import tushare as ts
from utils.confutils import ConfUtil

# 提取token
token = ConfUtil().get_tokens('tushare')
pro = ts.pro_api(token)
# 下载申万行业代码（2021年版本）
df = []
for src in ['SW2021']: #2021版本在2020年7月发布
    #获取申万行业列表
    df_tmp = pro.index_classify(src=src)
    df.append(df_tmp)
df = pd.concat(df,axis=0)
# 存到.\Data\RawData\index_sw_class.feather文件中
tofile = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) + "\\Data\\RawData\\index_sw_class.feather"
df.to_feather(tofile)